from django.contrib import admin
from django.urls import reverse
from .models import *

# Register your models here.


@admin.register(JSTOpen)
class JSTOpenAdmin(admin.ModelAdmin):
    save_on_top = True
    list_display = ['urls', 'app_key', 'app_secret', 'data_start_time']


@admin.register(JSTOpenImage)
class JSTOpenImageAdmin(admin.ModelAdmin):
    list_display = ['name', 'number', 'paths', 'create_time', 'update_time']


@admin.register(IpoImage)
class IpoImageAdmin(admin.ModelAdmin):
    list_display = ['username', 'image_name', 'net50_sim', 'time_net50_consuming',
                    'net101_sim', 'time_net101_consuming', 'accuracy', 'create_time']


@admin.register(Feature)
class FeatureAdmin(admin.ModelAdmin):
    list_display = ['model', 'update_time']


admin.site.site_header = '精诚智以图识图小工具'